from torchstat import stat
# from torchsummary import summary
from nets.unet import Unet
from nets.UNetwDYMicroBlock.unetwmicronet import Unetwmicroblock
from nets.UNetwDYMicroBlock import cfgs

INPUT_SIZE = (4, 224, 224)

model = Unet()
stat(model, INPUT_SIZE)

model = Unetwmicroblock(cfgs)
stat(model, INPUT_SIZE)
